from tensorflow.examples.tutorials.mnist import input_data
import tensorflow as tf
import os as os
os.environ["CUDA_VISIBLE_DEVICES"] = ""

tf.logging.set_verbosity(tf.logging.ERROR)
mnist = input_data.read_data_sets("/tmp/data/", one_hot=True)

print("Training data size:",mnist.train.num_examples)
print("Validate data size:",mnist.validation.num_examples)
print("Test data size:",mnist.test.num_examples)

print("Example train data:",mnist.train.images[0])
print("Example train labels:",mnist.train.labels[0])

batch_size = 100
xs,ys = mnist.train.next_batch(batch_size)
print("x shape:",xs)
print("y shape:",ys)